package com.ruoyi.framework.web.service;

import com.ruoyi.common.core.domain.entity.SysUser;
import com.ruoyi.common.core.domain.model.LoginUser;
import com.ruoyi.common.enums.UserStatus;
import com.ruoyi.common.exception.ServiceException;
import com.ruoyi.common.utils.MessageUtils;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.system.domain.SysPost;
import com.ruoyi.system.domain.SysUserPost;
import com.ruoyi.system.mapper.SysUserPostMapper;
import com.ruoyi.system.service.ISysUserService;
import com.ruoyi.system.service.impl.SysPostServiceImpl;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.security.core.userdetails.UserDetails;
import org.springframework.security.core.userdetails.UserDetailsService;
import org.springframework.security.core.userdetails.UsernameNotFoundException;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.HashSet;
import java.util.List;
import java.util.stream.Collectors;

/**
 * 用户验证处理
 *
 * @author ruoyi
 */
@Service
public class UserDetailsServiceImpl implements UserDetailsService {
	private static final Logger log = LoggerFactory.getLogger(UserDetailsServiceImpl.class);

	@Resource
	private SysUserPostMapper sysUserPostMapper;
	@Resource
	private ISysUserService userService;

	@Resource
	private SysPasswordService passwordService;

	@Resource
	private SysPermissionService permissionService;

	@Resource
	private SysPostServiceImpl sysPostService;

	@Override
	public UserDetails loadUserByUsername(String username) throws UsernameNotFoundException {
		SysUser user = userService.selectUserByUserName(username);
		if (StringUtils.isNull(user)) {
			log.info("登录用户：{} 不存在.", username);
			throw new ServiceException(MessageUtils.message("user.not.exists"));
		} else if (UserStatus.DELETED.getCode().equals(user.getDelFlag())) {
			log.info("登录用户：{} 已被删除.", username);
			throw new ServiceException(MessageUtils.message("user.password.delete"));
		} else if (UserStatus.DISABLE.getCode().equals(user.getStatus())) {
			log.info("登录用户：{} 已被停用.", username);
			throw new ServiceException(MessageUtils.message("user.blocked"));
		}

		passwordService.validate(user);

		/*身份集合*/
		HashSet<String> posts = new HashSet<>();
		List<SysUserPost> sysUserPostList = sysUserPostMapper.getAllByUserId(user.getUserId());
		List<SysPost> postList = sysUserPostList.stream().map(up -> sysPostService.selectPostById(up.getPostId())).collect(Collectors.toList());
		for (SysPost post : postList) {
			posts.add(post.getPostName());
		}
		List<Long> postIds = postList.stream().map(SysPost::getPostId).collect(Collectors.toList());
		user.setPostIds(postIds);
		user.setPosts(posts);

		return createLoginUser(user);
	}

	public UserDetails createLoginUser(SysUser user) {
		return new LoginUser(user.getUserId(), user.getDeptId(), user, permissionService.getMenuPermission(user), user.getPosts(), user.getPostIds());
	}
}
